# base image
FROM python:3.12-slim-bookworm AS base

WORKDIR /app/api

# Install uv
ENV UV_VERSION=0.6.14
ENV UV_DEFAULT_INDEX=https://mirrors.aliyun.com/pypi/simple/
# ENV LD_LIBRARY_PATH=/app/api;./
# if you located in China, you can use aliyun mirror to speed up
# RUN pip install --no-cache-dir poetry==${POETRY_VERSION} -i https://mirrors.aliyun.com/pypi/simple/

RUN pip install --no-cache-dir uv==${UV_VERSION} -i https://mirrors.aliyun.com/pypi/simple/

FROM base AS packages

RUN apt-get update \
    && apt-get install -y --no-install-recommends gcc g++ libc-dev libffi-dev libgmp-dev libmpfr-dev libmpc-dev 
    

# Install Python dependencies
COPY pyproject.toml uv.lock ./
RUN uv lock
RUN uv sync --locked
# RUN uv run pip install nls-1.1.0-py3-none-any.whl
# RUN poetry install --sync --no-cache --no-root

# production stage
FROM base AS production

ENV FLASK_APP=app.py
ENV EDITION=SELF_HOSTED
ENV DEPLOY_ENV=PRODUCTION
ENV CONSOLE_API_URL=http://127.0.0.1:5006
ENV CONSOLE_WEB_URL=http://127.0.0.1:3000
ENV SERVICE_API_URL=http://127.0.0.1:5006
ENV APP_WEB_URL=http://127.0.0.1:3000

EXPOSE 5006

# set timezone
ENV TZ=UTC

WORKDIR /app/api

RUN \
    apt-get update \
    # Install dependencies
    && apt-get install -y --no-install-recommends \
    # basic environment
    curl nodejs libgmp-dev libmpfr-dev libmpc-dev \
    # For Security
    expat libldap-2.5-0 perl libsqlite3-0 zlib1g \
    # install a chinese font to support the use of tools like matplotlib
    fonts-noto-cjk \
    # install a package to improve the accuracy of guessing mime type and file extension
    media-types \
    # install libmagic to support the use of python-magic guess MIMETYPE
    libmagic1 \
    # install libGL
    libgl1-mesa-glx \
    libglib2.0-0 \
    libsm6 \
    libxrender1 \
    libxext6 \
    # ffmpeg\
    && apt-get autoremove -y \
    && rm -rf /var/lib/apt/lists/*

# Copy Python environment and packages
ENV VIRTUAL_ENV=/app/api/.venv
COPY --from=packages ${VIRTUAL_ENV} ${VIRTUAL_ENV}
ENV PATH="${VIRTUAL_ENV}/bin:${PATH}"
# ENV NLTK_DATA=https://mirrors.ustc.edu.cn/nltk_data/
# # Download nltk data
# RUN python -c "import nltk; nltk.download('punkt'); nltk.download('averaged_perceptron_tagger')"

# ENV TIKTOKEN_CACHE_DIR=/app/api/.tiktoken_cache

# RUN python -c "import tiktoken; tiktoken.encoding_for_model('gpt2')"

# Copy source code
COPY . /app/api/


# Copy entrypoint
COPY docker/entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh

ARG COMMIT_SHA
ENV COMMIT_SHA=${COMMIT_SHA}

ENTRYPOINT ["/bin/bash", "/entrypoint.sh"]
